package com.kehao.leetcode.basic.array;

import org.junit.Test;

/**
 * 买卖股票的最佳时机 II
 * https://leetcode-cn.com/leetbook/read/top-interview-questions-easy/x2zsx1/
 * 给定一个数组，它的第 i 个元素是一支给定股票第 i 天的价格。
 *
 * 设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易（多次买卖一支股票）。
 *
 * 注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。
 *
 *
 * 示例 1:
 *
 * 输入: [7,1,5,3,6,4]
 * 输出: 7
 * 解释: 在第 2 天（股票价格 = 1）的时候买入，在第 3 天（股票价格 = 5）的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
 *    随后，在第 4 天（股票价格 = 3）的时候买入，在第 5 天（股票价格 = 6）的时候卖出, 这笔交易所能获得利润 = 6-3 = 3 。
 * 示例 2:
 *
 * 输入: [1,2,3,4,5]
 * 输出: 4
 * 解释: 在第 1 天（股票价格 = 1）的时候买入，在第 5 天 （股票价格 = 5）的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
 *    注意你不能在第 1 天和第 2 天接连购买股票，之后再将它们卖出。
 *     因为这样属于同时参与了多笔交易，你必须在再次购买前出售掉之前的股票。
 * 示例 3:
 *
 * 输入: [7,6,4,3,1]
 * 输出: 0
 * 解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。
 *
 */

public class ArraySolution2 {

    //利用动态规划
    public int maxProfit01(int[] prices) {
        int length = prices.length;
        int[] keepStock = new int[length];
        int[] noStock = new int[length];
        keepStock[0] = -prices[0];
        noStock[0] = 0;
        for (int i = 1; i < length; i++) {
            keepStock[i] = keepStock[i-1]>noStock[i-1]-prices[i]?keepStock[i-1]:noStock[i-1]-prices[i];
            noStock[i] = noStock[i-1]>keepStock[i-1]+prices[i]?noStock[i-1]:keepStock[i-1]+prices[i];
        }
        return keepStock[length-1]>noStock[length-1]?keepStock[length-1]:noStock[length-1];
    }

    //基于上面的动态规划方法进行优化，因为今天的钱数只和昨天有关 ，可以没有必要采用数组
    public int maxProfit02(int[] prices) {
        int length = prices.length;
        int keepStock= -prices[0];
        int noStock = 0;
        for (int i = 1; i < length; i++) {
            keepStock = keepStock>noStock-prices[i]?keepStock:noStock-prices[i];
            noStock = noStock>keepStock+prices[i]?noStock:keepStock+prices[i];
        }
        return noStock;
    }

    //贪心算法
    public int maxProfit03(int[] prices) {
        int earn = 0;
        for (int i = 1; i < prices.length; i++) {
            if(prices[i]>prices[i-1]){
                earn += prices[i]-prices[i-1];
            }
        }
        return earn;
    }

    public int maxProfit(int[] prices) {
        return maxProfit03(prices);
    }


    @Test
    public void myTest01(){
        int[] prices = {7,1,5,3,6,4};
        System.out.println(maxProfit(prices));
    }

    @Test
    public void myTest02(){
        int[] prices = {1,2,3,4,5};
        System.out.println(maxProfit(prices));
    }

    @Test
    public void myTest03(){
        int[] prices = {7,6,4,3,1};
        System.out.println(maxProfit(prices));
    }

}
